package com.aqie.easy.math;

import java.util.Arrays;

/**
 * 66 由整数组成的非空数组所表示的非负整数，在该数的基础上加一
 */
public class PlusOne {
    /**
     * 1ms
     * @param digits
     * @return
     */
    public static int[] plusOne(int[] digits) {
        int N = digits.length - 1;  // 1
        for (int i = N; i >= 0; i--){
            if (digits[i] == 9 && i == 0){
                int[] ans =  new int[N+2];
                ans[0] = 1;
                return ans;
            }
            if (digits[i] == 9) {
                digits[i] = 0;
            }else {
                digits[i] += 1;
                break;
            }
        }
        return digits;
    }

    public static void main(String[] args) {
        // int[] a = {1,2,3};
        int[] a = {9};
        // int[] a = {8, 9};

        System.out.println(Arrays.toString(plusOne(a)));
    }
}
